#!/usr/bin/python3

import secrets_info

import openai
import time

# 设置API密钥
openai.api_key = secrets_info.GPT4_API_KEY
openai.api_base = secrets_info.GPT4_URL_BASE

question = [
"罗杰有五个网球,他又买了两盒网球,每盒有3个网球,请问他现在总共有多少个网球?",
"食堂总共有23个苹果,如果他们用掉20个苹果,然后又买了6个苹果,请问现在食堂总共有多少个苹果?",
"杂耍者可以杂耍16个球。其中一半的球是高尔夫球,其中一半的高尔夫球是蓝色的。请问总共有多少个蓝色高尔夫球?",
"艾米需要4分钟才能爬到滑梯顶部,她花了1分钟才滑下来,滑梯将在15分钟后关闭,请问在关闭之前她能滑多少次?"
]

for var in question:
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo-0613",
        messages=[
            {"role": "user", "content": var}
        ]
    )
    print("Generated text:\n", response.choices[0].message['content'])
    print("================================")

